package com.atguigu.producer;

import lombok.extern.slf4j.Slf4j;
import org.apache.kafka.clients.CommonClientConfigs;
import org.apache.kafka.clients.producer.KafkaProducer;
import org.apache.kafka.clients.producer.ProducerConfig;
import org.apache.kafka.clients.producer.ProducerRecord;
import org.apache.kafka.common.serialization.StringSerializer;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import java.util.Properties;
import java.util.concurrent.ExecutionException;

/**
 * producer执行顺序：拦截器-->序列化器-->分区器
 *
 * @author code1997
 */
@Slf4j
public class MyProducer {

    KafkaProducer<String, String> producer;

    /**
     * 三个配置常量类：
     * CommonClientConfigs:公共的配置信息.
     * ProducerConfig:生产者的配置信息.
     * ConsumerConfig:消费者的配置信息.
     */
    @Before
    public void initProperty() {
        //1 创建kafka生产者的配置信息
        Properties properties = new Properties();
        //2 指定连接的kafka集群地址信息
        properties.put(CommonClientConfigs.BOOTSTRAP_SERVERS_CONFIG, "hadoop02:9092");
        //3 ack的应答级别
        properties.put(ProducerConfig.ACKS_CONFIG, "all");
        //4 重试次数
        properties.put(ProducerConfig.RETRIES_CONFIG, 3);
        //5 批次大小，单位是字节，字节16k
        properties.put(ProducerConfig.BATCH_SIZE_CONFIG, 16384);
        //6 等待时间
        properties.put(ProducerConfig.LINGER_MS_CONFIG, 1);
        //7 RecordAccumulator 缓冲区大小(32M)
        properties.put(ProducerConfig.BUFFER_MEMORY_CONFIG, 33554432);
        //8 k,v序列化方式
        properties.put(ProducerConfig.KEY_SERIALIZER_CLASS_CONFIG, StringSerializer.class);
        properties.put(ProducerConfig.VALUE_SERIALIZER_CLASS_CONFIG, StringSerializer.class);
        producer = new KafkaProducer<>(properties);
    }

    /**
     * 注意发送数据是一批批的发送，因此可能不是很平均
     * 回调函数存在两个参数：metadata, exception
     * 注意ProducerRecord的参数
     */
    @Test
    public void sendByAsync() {
        for (int i = 0; i < 10; i++) {
            producer.send(new ProducerRecord<>("kafka-test", "code1997", "code1997" + i), (metadata, exception) -> {
                //回调函数
                if (exception == null) {
                    log.info("发送消息成功,topic:{},offset:{},partition:{}", metadata.offset(), metadata.offset(), metadata.partition());
                } else {
                    //说明发送消息失败
                    log.error("发送消息发生异常,topic:{},offset:{},partition:{}", metadata.offset(), metadata.offset(), metadata.partition(), exception);
                }
            });
        }
    }

    /**
     * 发送之后调用get方法则是同步调用,虽然存在两个线程--main线程，sender线程，但是一旦使用同步发送，main线程发送之后会阻塞。
     */
    @Test
    public void sendBySync() {
        for (int i = 0; i < 100000; i++) {
            try {
                producer.send(new ProducerRecord<>("kafka-test", "code1997", "code1997" + i)).get();
            } catch (InterruptedException | ExecutionException e) {
                log.error("发送消息到kafka出现异常", e);
            }
        }
    }

    @After
    public void closeResource() {
        producer.close();
    }
}
